package com.ht.api.call.func.impl.task;

import com.ht.api.bean.AjaxResult;
import com.ht.api.bean.LoginBean;
import com.ht.api.call.func.IFunctionCall;
import com.ht.api.db.Db;
import com.ht.api.db.Entity;
import com.ht.api.db.exception.HtException;
import com.ht.api.db.task.Plan;
import com.ht.api.db.task.TaskInfo;

import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import lombok.extern.slf4j.Slf4j;

/**
 * 业务接口：修改执行计划
 * @author asq
 * @createTime 2025年2月8日 21:14:29
 */
@Slf4j
public class TaskChangeFunctionCall implements IFunctionCall {

	@Override
	public String getName() {
		return "修改执行计划";
	}

	@Override
	public JSONObject execute(LoginBean me, JSONObject param, AjaxResult result) {
		// 获取任务
		Long taskId = param.getLong("taskId");
		if (taskId == null) {
			throw new HtException("请指定任务ID参数：taskId");
		}
		Entity task = Db.createSqlParser("db_task").findById(taskId);
		if (task == null) {
			throw new HtException("任务不存在，ID：" + taskId);
		}
		if (TaskInfo.RUNNING.equals(task.getStr("status"))) {
			throw new HtException("请先停止任务再修改执行计划");
		}
		
		// 获取新的执行计划
		Plan plan = param.getBean("plan", Plan.class);
		if (plan == null) {
			throw new HtException("请指定执行计划：plan");
		}
		
		// 修改执行计划
		TaskInfo taskInfo = TaskInfo.load(task);
		plan.setPlan(taskInfo);
		taskInfo.setAttempts(0); // 将当前执行计划下的执行次数归0，再次执行任务时，便可以从头开始执行
		Db.insert(taskInfo.toEntry());
		
		log.info("[TaskChangeFunctionCall]修改执行计划,taskId:{}", taskId);
		
		return JSONUtil.createObj();
	}

}
